from enum import Enum

single_version = "v20240521"
double_version = "v20240614"

# spark参数
result_root_path = "~/result/"
spark_save_mode = "overwrite"
spark_partition_columns = ["p_date", "fault_code", "system_type", "version"]

# sagemaker参数
sagemaker_write_file = "result.csv"


class SystemType(Enum):
    SINGLE = "single_system"
    DOUBLE = "double_system"


class FaultCode(Enum):
    FRIDGE_ZHILENG = "refrigeratingSystem_Fault"
    AIRDOOR = "airDoor_Fault"


class ProtoCol(Enum):
    HTTP = "http"
    MQTT = "mqtt"


class TableName(Enum):
    HTTP = "ods.iot_fridge"
    MQTT = "ods.iot_fridge_mqtt"
